1398C - Good Subarrays - CodeForces Solution

from math import ceil
import os
import sys
from io import BytesIO, IOBase
 
 
BUFSIZE: int = 8192
 
 
class FastIO(IOBase):
    newlines = 0
 
    def __init__(self, file):
        self._fd = file.fileno()
        self.buffer = BytesIO()
        self.writable = "x" in file.mode or "r" not in file.mode
        self.write = self.buffer.write if self.writable else None
 
    def read(self):
        while True:
            b = os.read(self._fd, max(os.fstat(self._fd).st_size, BUFSIZE))
            if not b:
                break
            ptr = self.buffer.tell()
            self.buffer.seek(0, 2), self.buffer.write(b), self.buffer.seek(ptr)
        self.newlines = 0
        return self.buffer.read()
 
    def readline(self):
        while self.newlines == 0:
            b = os.read(self._fd, max(os.fstat(self._fd).st_size, BUFSIZE))
            self.newlines = b.count(b"\n") + (not b)
            ptr = self.buffer.tell()
            self.buffer.seek(0, 2), self.buffer.write(b), self.buffer.seek(ptr)
        self.newlines -= 1
        return self.buffer.readline()
 
    def flush(self):
        if self.writable:
            os.write(self._fd, self.buffer.getvalue())
            self.buffer.truncate(0), self.buffer.seek(0)
 
 
class IOWrapper(IOBase):
    def __init__(self, file):
        self.buffer = FastIO(file)
        self.flush = self.buffer.flush
        self.writable = self.buffer.writable
        self.write = lambda s: self.buffer.write(s.encode("ascii"))
        self.read = lambda: self.buffer.read().decode("ascii")
        self.readline = lambda: self.buffer.readline().decode("ascii")
 
 
def print(*args, **kwargs):
    
    sep, file = kwargs.pop("sep", " "), kwargs.pop("file", sys.stdout)
    at_start = True
    for x in args:
        if not at_start:
            file.write(sep)
        file.write(str(x))
        at_start = False
    file.write(kwargs.pop("end", "\n"))
    if kwargs.pop("flush", False):
        file.flush()
 
 
sys.stdin, sys.stdout = IOWrapper(sys.stdin), IOWrapper(sys.stdout)
input = lambda: sys.stdin.readline().rstrip("\r\n")
 
 
if __name__ == "__main__":
    pass
def inpi():
    return(int(input()))
def inpl():
    return(list(map(int,input().split())))
def inps():
    s = input()
    return(list(s[:len(s)]))
def inpm():
    return(map(int,input().split()))
def inpst():
    return(set(map(int,input().split())))


    
def solve():
    n = inpi()
    a = [int(i)for i in input()]
    pre = [0]*n
    d = {0:1}
    res = 0
    for i in range(n):
        pre[i] = pre[i-1] + a[i]
                if pre[i] - i - 1 in d:
            res += (d[pre[i] - i - 1])
            d[pre[i] - i - 1] += 1
        else:
            d[pre[i] - i - 1] = 1
        print(res)
    
        
'''

'''

    
t = 1
t = inpi()
for _ in range(t):
    solve()